#!/usr/bin/env python
#-*- coding:utf-8 -*-
# author:wangfangfang
# datetime:2019/5/30
# software: PyCharm Community Edition
from selenium import webdriver
from selenium.webdriver.support.select import Select
import time,random
from common.login import Login
class Entrepreneur():
    # 调用登录模块，打开浏览器并登
    def __init__(self, role_a):
        self.driver = Login(role_a).login()
        time.sleep(2)
    def entrepreneur(self):
        # 定位企业客户元素，并点击元素
        self.driver.find_element_by_link_text('企业客户').click()
        time.sleep(1)
        # 定位企业客户元素，并点击元素
        self.driver.find_element_by_xpath('//*[@id="list-6"]/div/ul/li/a').click()
    def Company(self,company): #输入查询方法
        # 定位企业名称输入框元素，并输入值
        self.driver.find_element_by_name('companyName').send_keys(company)
        # 定位搜索元素，并点击元素
        self.driver.find_element_by_xpath('//*[@id="content"]/div[2]/div/button[1]').click()
    def Increased(self,newentname,newentcate,newentaddr,newentheader,newtel,newemail,ent): #新增企业方法
        # 定位新增元素，并点击
        self.driver.find_element_by_xpath('//*[@id="content"]/div[2]/div/button[2]').click()
        # 定位企业名称输入框元素，并输入值
        self.driver.find_element_by_id('newentname').send_keys(newentname)
        # 定位所属行业输入框元素，并输入值
        self.driver.find_element_by_id('newentcate').send_keys(newentcate)
        # 定位企业地址输入框元素，并输入值
        self.driver.find_element_by_id('newentaddr').send_keys(newentaddr)
        # 定位联系人输入框元素，并输入值
        self.driver.find_element_by_id('newentheader').send_keys(newentheader)
        # 定位联系电话输入框元素，并输入值
        self.driver.find_element_by_id('newtel').send_keys(newtel)
        # 定位邮箱输入框元素，并输入值
        self.driver.find_element_by_id('newemail').send_keys(newemail)
        # 定位QQ输入框元素，并输入值
        self.driver.find_element_by_xpath('//*[@id="enterprise_form"]/div[7]/input').send_keys(ent)
        # 定位添加元素，并点击
        self.driver.find_element_by_xpath('//*[@id="company-add"]/div/div/div[3]/button').click()
        time.sleep(1)
        # 定位确定元素，并点击
        self.driver.find_element_by_xpath('/html/body/div[10]/div/div/div[3]/button').click()
        time.sleep(1)
    def Modifier(self,entName,entCate,entAddr,entHeader,entTel,entEmail,entQq): #修改企业方法
        # 定位修改元素，并点击
        self.driver.find_element_by_xpath('//*[@id="enterpriseTb"]/tbody/tr[1]/td[9]/button').click()
        # 定位企业名称输入框元素，并输入值
        self.driver.find_element_by_id('entName').clear()
        self.driver.find_element_by_id('entName').send_keys(entName)
        # 定位所属行业输入框元素，并输入值
        self.driver.find_element_by_id('entCate').clear()
        self.driver.find_element_by_id('entCate').send_keys(entCate)
        # 定位企业地址输入框元素，并输入值
        self.driver.find_element_by_id('entAddr').clear()
        self.driver.find_element_by_id('entAddr').send_keys(entAddr)
        # 定位联系人输入框元素，并输入值
        self.driver.find_element_by_id('entHeader').clear()
        self.driver.find_element_by_id('entHeader').send_keys(entHeader)
        # 定位联系电话输入框元素，并输入值
        self.driver.find_element_by_id('entTel').clear()
        self.driver.find_element_by_id('entTel').send_keys(entTel)
        # 定位邮箱输入框元素，并输入值
        self.driver.find_element_by_id('entEmail').clear()
        self.driver.find_element_by_id('entEmail').send_keys(entEmail)
        # 定位QQ输入框元素，并输入值
        self.driver.find_element_by_id('entQq').clear()
        self.driver.find_element_by_id('entQq').send_keys(entQq)
        # 定位保存元素，并点击
        self.driver.find_element_by_xpath('//*[@id="company-modif"]/div/div/div[3]/button').click()
        time.sleep(2)
        # 定位确定元素，并点击
        self.driver.find_element_by_xpath('/html/body/div[10]/div/div/div[3]/button').click()
if __name__ == '__main__':
    a = Entrepreneur("超级管理员")
    a.entrepreneur()
    a.Company("")
    a.Increased("天元公司","土木","天堂","你","13245678907","8565434@qq.com","12345")
    a.Modifier("天元海公司","土木","天堂","你","13245678907","8565434@qq.com","12345")
